package liming.ai.tongyi;

import liming.ai.abstracts.AbstractMessage;
import liming.tool.json.annotation.JsonMark;

public class TongYiMessage extends AbstractMessage {
    @JsonMark
    public String role;
    @JsonMark
    public String content;

    public TongYiMessage(Role role, String content) {
        this.role = role.toString();
        this.content = content;
    }

    public enum Role {
        USER,
        ASSISTANT,
        SYSTEM;

        @Override
        public String toString() {
            return super.toString().toLowerCase();
        }
    }
}
